package com.micai.boss.base.encrypt;

import sun.misc.BASE64Decoder;
import sun.misc.BASE64Encoder;

import javax.crypto.BadPaddingException;
import javax.crypto.Cipher;
import javax.crypto.IllegalBlockSizeException;
import javax.crypto.NoSuchPaddingException;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.security.*;
import java.security.interfaces.RSAPrivateKey;
import java.security.interfaces.RSAPublicKey;
import java.security.spec.InvalidKeySpecException;
import java.security.spec.PKCS8EncodedKeySpec;
import java.security.spec.X509EncodedKeySpec;

/**
 * 描述：
 * <p>
 *
 * @author: 赵新国
 * @date: 2018/4/23 15:47
 */
public class RSASecurityUtils {

    // RSA 公钥
    public static final String DEFAULT_PUBLIC_KEY = "MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQCALoN8yEGn7okKYyKg6b1g+DAk3DFt0GXrEls8\n" +
            "KbLI6fgD7BhjSk9D9oGF1BkViKixRFaTUzTqFzs5kFQAgtulh1VAJ/4DSAT6Ara1eAw8D+J9fZnR\n" +
            "Gfe/7xXCiAUGQfDBH2Orq2gQQIVjNZFtgPtN2ozt80vpHh3qdiQ6O6ApawIDAQAB";
    // RSA 私钥
    //RSA PKCS#8编码格式的私钥
    public static final String DEFAULT_PKCS8_PRIVATE_KEY = "MIICdQIBADANBgkqhkiG9w0BAQEFAASCAl8wggJbAgEAAoGBAIAug3zIQafuiQpjIqDpvWD4MCTc\n" +
            "MW3QZesSWzwpssjp+APsGGNKT0P2gYXUGRWIqLFEVpNTNOoXOzmQVACC26WHVUAn/gNIBPoCtrV4\n" +
            "DDwP4n19mdEZ97/vFcKIBQZB8MEfY6uraBBAhWM1kW2A+03ajO3zS+keHep2JDo7oClrAgMBAAEC\n" +
            "gYAhNmwhZ+4DKj5d76PoV8gXGRe4ZUl1b7sItr/nDdLiViVKNuF20KlLLdMPrw400y0KiW95SuT6\n" +
            "Brz0xF3PqBhrIfOkRcn6eJXiH5HwQvGpGlfrWlHb+bL7w3Xy7ykI0Y+biKivGxGDeU6x8rmIpbv+\n" +
            "qcHZx9mEVskocqXMDWZPcQJBAOUp5RUlIWJoVMvsIaJhh0Yi3KaQLfbGOe64yAzniJhxlU/e8Rbk\n" +
            "uWk1hsXMLEtuLqJrFQXyMxvzsEkjwSo3vFkCQQCPMUbw7FN7zjQYK/QrgkXFUOMQVkrn/e1xqUU9\n" +
            "UguD5ECJIHqpuvH48XfrtydCiWwjm0I8OMxFShf8shlZfotjAkAj3oTe5GrvTklHifRRyFY2IirL\n" +
            "X3HyWGCk/wNBEdSNl+m5ELpHzRFAaIJp32VngR2Hip9FyP06+M+1AwWvJ/4xAkArD2qB8M96H5xS\n" +
            "2ITowJgyqgxeUVTdRuMslvaXZiDIXM97bST1/Z3eet7RJ5bKz4CRhJL6gSoMfCeSegxdeUx5AkAh\n" +
            "Yq0CDcqJTLvHgUQ3rMeqXptGObKq4UbsXywk1yY68n64mfG8fDkxtSYrzdNcAcRlJQ20MPsaCjQc\n" +
            "JFImrCAs";

    // 非对称加密密钥算法
    private static final String KEY_ALGORITHM = "RSA";
    // 私钥
    private static RSAPrivateKey privateKey;
    // 公钥
    private static RSAPublicKey publicKey;
    // 字节数据转字符串专用集合
    private static final char[] HEX_CHAR= {'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c', 'd', 'e', 'f'};

    public static RSAPrivateKey getPrivateKey() {
        return privateKey;
    }

    public static RSAPublicKey getPublicKey() {
        return publicKey;
    }

    /**
     * 随机生成密钥对
     */
    public static void genKeyPair(){
        KeyPairGenerator keyPairGen= null;
        try {
            keyPairGen= KeyPairGenerator.getInstance(KEY_ALGORITHM);
        } catch (NoSuchAlgorithmException e) {
            e.printStackTrace();
        }
        keyPairGen.initialize(1024, new SecureRandom());
        KeyPair keyPair= keyPairGen.generateKeyPair();
        privateKey= (RSAPrivateKey) keyPair.getPrivate();
        publicKey= (RSAPublicKey) keyPair.getPublic();
    }

    /**
     * 从文件中输入流中加载公钥
     * @param in 公钥输入流
     * @throws Exception 加载公钥时产生的异常
     */
    public static void loadPublicKey(InputStream in) throws Exception{
        try {
            BufferedReader br= new BufferedReader(new InputStreamReader(in));
            String readLine= null;
            StringBuilder sb= new StringBuilder();
            while((readLine= br.readLine())!=null){
                if(readLine.charAt(0)=='-'){
                    continue;
                }else{
                    sb.append(readLine);
                    sb.append('\r');
                }
            }
            loadPublicKey(sb.toString());
        } catch (IOException e) {
            throw new Exception("公钥数据流读取错误");
        } catch (NullPointerException e) {
            throw new Exception("公钥输入流为空");
        }
    }

    /**
     * 从字符串中加载公钥
     * @param publicKeyStr 公钥数据字符串
     * @throws Exception 加载公钥时产生的异常
     */
    public static void loadPublicKey(String publicKeyStr) throws Exception{
        try {
            BASE64Decoder base64Decoder= new BASE64Decoder();
            byte[] buffer= base64Decoder.decodeBuffer(publicKeyStr);
            KeyFactory keyFactory= KeyFactory.getInstance(KEY_ALGORITHM);
            X509EncodedKeySpec keySpec= new X509EncodedKeySpec(buffer);
            publicKey= (RSAPublicKey) keyFactory.generatePublic(keySpec);
        } catch (NoSuchAlgorithmException e) {
            throw new Exception("无此算法");
        } catch (InvalidKeySpecException e) {
            throw new Exception("公钥非法");
        } catch (IOException e) {
            throw new Exception("公钥数据内容读取错误");
        } catch (NullPointerException e) {
            throw new Exception("公钥数据为空");
        }
    }

    /**
     * 从文件中加载私钥
     * @param in keyFileName 私钥文件名
     * @return 是否成功
     * @throws Exception
     */
    public static void loadPrivateKey(InputStream in) throws Exception{
        try {
            BufferedReader br= new BufferedReader(new InputStreamReader(in));
            String readLine= null;
            StringBuilder sb= new StringBuilder();
            while((readLine= br.readLine())!=null){
                if(readLine.charAt(0)=='-'){
                    continue;
                }else{
                    sb.append(readLine);
                    sb.append('\r');
                }
            }
            loadPrivateKey(sb.toString());
        } catch (IOException e) {
            throw new Exception("私钥数据读取错误");
        } catch (NullPointerException e) {
            throw new Exception("私钥输入流为空");
        }
    }

    /**
     * 从字符串中加载私钥
     * @param privateKeyStr 私钥数据字符串
     * @throws Exception 加载私钥时产生的异常
     */
    public static void loadPrivateKey(String privateKeyStr) throws Exception{
        try {
            BASE64Decoder base64Decoder= new BASE64Decoder();
            byte[] buffer= base64Decoder.decodeBuffer(privateKeyStr);
            PKCS8EncodedKeySpec keySpec= new PKCS8EncodedKeySpec(buffer);
            KeyFactory keyFactory= KeyFactory.getInstance(KEY_ALGORITHM);
            privateKey= (RSAPrivateKey) keyFactory.generatePrivate(keySpec);
        } catch (NoSuchAlgorithmException e) {
            throw new Exception("无此算法");
        } catch (InvalidKeySpecException e) {
            throw new Exception("私钥非法");
        } catch (IOException e) {
            throw new Exception("私钥数据内容读取错误");
        } catch (NullPointerException e) {
            throw new Exception("私钥数据为空");
        }
    }

    /**
     * 加密过程
     * @param publicKey 公钥
     * @param plainTextData 明文数据
     * @return byte[]
     * @throws Exception 加密过程中的异常信息
     */
    public static byte[] encrypt(RSAPublicKey publicKey, byte[] plainTextData) throws Exception{
        if(publicKey== null){
            throw new Exception("加密公钥为空, 请设置");
        }
        Cipher cipher= null;
        try {
            cipher= Cipher.getInstance("RSA/ECB/PKCS1Padding");
            cipher.init(Cipher.ENCRYPT_MODE, publicKey);
            byte[] output= cipher.doFinal(plainTextData);
            return output;
        } catch (NoSuchAlgorithmException e) {
            throw new Exception("无此加密算法");
        } catch (NoSuchPaddingException e) {
            e.printStackTrace();
            return null;
        }catch (InvalidKeyException e) {
            throw new Exception("加密公钥非法,请检查");
        } catch (IllegalBlockSizeException e) {
            throw new Exception("明文长度非法");
        } catch (BadPaddingException e) {
            throw new Exception("明文数据已损坏");
        }
    }

    /**
     * 解密过程
     * @param privateKey 私钥
     * @param cipherData 密文数据
     * @return 明文
     * @throws Exception 解密过程中的异常信息
     */
    public static byte[] decrypt(RSAPrivateKey privateKey, byte[] cipherData) throws Exception{
        if (privateKey== null){
            throw new Exception("解密私钥为空, 请设置");
        }
        Cipher cipher= null;
        try {
            cipher= Cipher.getInstance("RSA/ECB/PKCS1Padding");
            cipher.init(Cipher.DECRYPT_MODE, privateKey);
            byte[] output = cipher.doFinal(cipherData);
            return output;
        } catch (NoSuchAlgorithmException e) {
            throw new Exception("无此解密算法");
        } catch (NoSuchPaddingException e) {
            e.printStackTrace();
            return null;
        }catch (InvalidKeyException e) {
            throw new Exception("解密私钥非法,请检查");
        } catch (IllegalBlockSizeException e) {
            throw new Exception("密文长度非法");
        } catch (BadPaddingException e) {
            throw new Exception("密文数据已损坏");
        }
    }

    /**
     * 字节数据转十六进制字符串
     * @param data 输入数据
     * @return 十六进制内容
     */
    public static String byteArrayToString(byte[] data){
        StringBuilder stringBuilder= new StringBuilder();
        for (int i=0; i<data.length; i++){
            //取出字节的高四位 作为索引得到相应的十六进制标识符 注意无符号右移
            stringBuilder.append(HEX_CHAR[(data[i] & 0xf0)>>> 4]);
            //取出字节的低四位 作为索引得到相应的十六进制标识符
            stringBuilder.append(HEX_CHAR[(data[i] & 0x0f)]);
            if (i<data.length-1){
                stringBuilder.append(' ');
            }
        }
        return stringBuilder.toString();
    }

    /**
     * 明文公钥加密 <br/>
     * @param data 输入的明文
     * @param key 公钥
     * @return 密文
     */
    public static String publicKeyEncrypt(String data, String key){
        String encrypt_ = null;
        try {
            // 加载公钥
            loadPublicKey(key);

            // 公钥加密
            byte[] cipher = encrypt(getPublicKey(), data.getBytes());
            return (new BASE64Encoder()).encode(cipher);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return encrypt_;
    }

    /**
     * 密文私钥解密 <br/>
     * @param data 输入的密文
     * @param key 私钥
     * @return 密文
     */
    public static String privateKeyEncrypt(String data, String key){
        String decrypt_ = null;
        try {
            // 加载公钥
            loadPrivateKey(key);

            // 私钥解密
            byte[] plainText = decrypt(getPrivateKey(), (new BASE64Decoder()).decodeBuffer(data));
            decrypt_ = new String(plainText);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return decrypt_;
    }


//	public static void main(String[] args){
////        测试明文字符串
//        String encryptStr= "10000,13800138000,lueSGJZetyySpUndWjMBEg==";
//        try {
//        	System.out.println("输入的明文原文: " + encryptStr);
//
//        	String encrypt_ = publicKeyEncrypt(encryptStr, DEFAULT_PUBLIC_KEY);
//        	System.out.println("密文原文: "+ encrypt_);
//
//        	String decrypt_ = privateKeyEncrypt(encrypt_, DEFAULT_PKCS8_PRIVATE_KEY);
//        	System.out.println("解密后的明文原文: "+ decrypt_);
//        	System.out.println("=============================================================================================");

//        	// 指定的密文解密测试
//        	String ios_ = "b6faef6c9e41d06a9380f40b5165cf5d83bfbdacda2dff11f411f159436099eb731f8fa2c2c6cd41de7786b3307fc9e2e56984682b366c550ad5fb2ca24700ec8172e0ff3c519458592950938cdc06dd3622ad4982fb0d4f8010ae47970448c0f7a6083c9106c31ee13c946bde18efbc16dbbccae5da37d74118182f2915ef84";
//            String android_ = "QmGyQMeX/nwwSXKU8W/qN7uDmxUQAe4+31OvtUVPZH/Swquwex+v7FgRxCVViowiKj9bdH0SZda8DuljEsF7NF7nX4DL2rP5pH0KbgW1ddYNYBuVBtcRkvob0G5SDzQo3lueRRwhuz7N8LZc6+2irEoebV4Y30AEC6wMgc1Ku2A=";
//            //密文原文
//            System.out.println("指定密文原文ios_:"+ ios_);
//            decrypt_ = privateKeyEncrypt(ios_, DEFAULT_PKCS8_PRIVATE_KEY);
//        	System.out.println("解密指定后的明文原文ios_: "+ decrypt_);
//        	System.out.println("=============================================================================================");
//        	//密文原文
//        	System.out.println("密文指定原文android_:"+ android_);
//        	String decrypt_ = privateKeyEncrypt(android_, DEFAULT_PKCS8_PRIVATE_KEY);
//        	System.out.println("解密指定后的明文原文android_: "+ decrypt_);
//        } catch (Exception e) {
//            System.err.println(e.getMessage());
//        }
//    }
}
